# -*- coding: utf-8 -*-
# @File     : detect_cfg.py
# @Author   : bingjia
# @Time     : 2020/9/22 9:31
# @Desc     : 检测配置文件
import os
from easydict import EasyDict as edict

ROOT_DIR = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))

__C = edict()
defect_cfg = __C

"""
检测设置
"""
"检测设备，cpu或者gpu 0 1 2"
__C.device = ''

"检测图片尺度"
__C.img_size = 640

"NMS iou阈值"
__C.iou_thres = 0.5

"对象置信度阈值"
__C.conf_thres = 0.4

__C.classes = None

"更新所有模型"
__C.update = False

"增强推理"
__C.augment = False

"保存结果文件"
__C.save_txt = False

"显示结果"
__C.view_img = False

__C.agnostic_nms = False

__C.weights = os.path.join(ROOT_DIR, 'weights', 'best.pt')

__C.output = os.path.join(ROOT_DIR, 'inference', 'output')

__C.source = os.path.join(ROOT_DIR, 'inference', 'images')
